// Copyright (c) Appeon
// The following has been modified by Appeon under compliance with the Apache 2.0 license from the original work

// Copyright(c) 2002-2019, Npgsql
// Permission to use, copy, modify, and distribute this software and its
// documentation for any purpose, without fee, and without a written agreement
// is hereby granted, provided that the above copyright notice and this
// paragraph and the following two paragraphs appear in all copies.
//
// IN NO EVENT SHALL NPGSQL BE LIABLE TO ANY PARTY FOR DIRECT, INDIRECT,
// SPECIAL, INCIDENTAL, OR CONSEQUENTIAL DAMAGES, INCLUDING LOST PROFITS,
// ARISING OUT OF THE USE OF THIS SOFTWARE AND ITS DOCUMENTATION, EVEN IF
// Npgsql HAS BEEN ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
//
// NPGSQL SPECIFICALLY DISCLAIMS ANY WARRANTIES, INCLUDING, BUT NOT LIMITED
// TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
// PURPOSE.THE SOFTWARE PROVIDED HEREUNDER IS ON AN "AS IS" BASIS, AND Npgsql
// HAS NO OBLIGATIONS TO PROVIDE MAINTENANCE, SUPPORT, UPDATES, ENHANCEMENTS,
// OR MODIFICATIONS.

using System;
using System.Text;
using JetBrains.Annotations;
using Npgsql;
using NpgsqlTypes;

namespace SnapObjects.Data.PostgreSql
{
    public class PostgreSqlRangeTypeMapping : PostgreSqlTypeMapping
    {
        public PostgreSqlRangeTypeMapping(
            [NotNull] string storeType, [NotNull] Type clrType, [NotNull] TypeMapping subtypeMapping)
            : base(storeType, clrType, GenerateNpgsqlDbType(subtypeMapping))
        {
            this.SubTypeMapping = subtypeMapping;
        }

        private TypeMapping SubTypeMapping { get; }

        protected override string GenerateNonNullSqlLiteral(object value)
        {
            var sb = new StringBuilder();

            sb.Append('\'');
            sb.Append(value);
            sb.Append("'::");
            sb.Append(this.DbTypeName);

            return sb.ToString();
        }

        private static NpgsqlDbType GenerateNpgsqlDbType([NotNull] TypeMapping subtypeMapping)
        {
            if (subtypeMapping is PostgreSqlTypeMapping npgsqlTypeMapping)
            {
                return NpgsqlDbType.Range | npgsqlTypeMapping.NpgsqlDbType;
            }

            var p = new NpgsqlParameter();

            p.DbType = subtypeMapping.DbType.Value;

            return NpgsqlDbType.Range | p.NpgsqlDbType;
        }

        public override TypeMapping Clone([NotNull] string dbTypeName, int? size)
        {
            return new PostgreSqlRangeTypeMapping(dbTypeName, this.ClrType, this.SubTypeMapping);
        }
    }
}
